

# python train.py \
#     --root_dir gaofen_aug \
#     --num_workers=8 \
#     --net_G=base_transformer_pos_s4_dd8_dedim8 \
#     --batch_size=8 \
#     --pretrained=pretrained/pos_s4_converted.pt \
#     --in_channels=4 \
#     --lr=1e-5 \
#     --eval_steps=1000 \

python train.py \
    --root_dir gaofen_aug \
    --num_workers=2 \
    --net_G=snunet \
    --batch_size=2 \
    --pretrained=pretrained/snunet-32-converted.pt \
    --in_channels=4 \
    --lr=1e-5 \
    --eval_steps=1000 \